package com.es.designmode.composite;

import cn.hutool.core.util.StrUtil;

import java.util.ArrayList;
import java.util.List;

/**
 * 定义有枝节点行为，用来存储子部件，相当于树枝
 * 实现 add 和 remove方法
 *
 * @author Will
 */
public class Composite extends Component {

    private List<Component> children = new ArrayList<Component>();

    public Composite(String name) {
        super(name);
    }

    @Override
    public void add(Component c) {
        children.add(c);
    }

    @Override
    public void remove(Component c) {
        children.remove(c);
    }

    @Override
    public void display(int depth) {
        System.out.println(StrUtil.repeat('-', depth) + getName());

        // 在这里使用了递归
        for (Component c : children) {
            c.display(depth + 2);
        }
    }
}
